#! /usr/bin/python
# -*- coding: utf-8 -*-


import wmi
import sys, time, platform


def get_system_info(os):
  """
  获取操作系统版本。
  """
  print
  print "Operating system:"
  if os == "Windows":
    c = wmi.WMI()
    for sys in c.Win32_OperatingSystem():
      print '\t' + "Version :\t%s" % sys.Caption.encode("GBK")
      print '\t' + "Vernum :\t%s" % sys.BuildNumber


def get_memory_info(os):
  """
  获取物理内存和虚拟内存。
  """
  print
  print "memory_info:"
  if os == "Windows":
    c = wmi.WMI()
    cs = c.Win32_ComputerSystem()
    pfu = c.Win32_PageFileUsage()
    MemTotal = int(cs[0].TotalPhysicalMemory) / 1024 / 1024
    print '\t' + "TotalPhysicalMemory :" + '\t' + str(MemTotal) + "M"
    # tmpdict["MemFree"] = int(os[0].FreePhysicalMemory)/1024
    SwapTotal = int(pfu[0].AllocatedBaseSize)
    print '\t' + "SwapTotal :" + '\t' + str(SwapTotal) + "M"
    # tmpdict["SwapFree"] = int(pfu[0].AllocatedBaseSize - pfu[0].CurrentUsage)


def get_disk_info(os):
  """
  获取物理磁盘信息。
  """
  print
  print "disk_info:"
  if os == "Windows":
    tmplist = []
    c = wmi.WMI()
    for physical_disk in c.Win32_DiskDrive():
      if physical_disk.Size:
        print '\t' + str(physical_disk.Caption) + ' :\t' + str(long(physical_disk.Size) / 1024 / 1024 / 1024) + "G"


def get_cpu_info(os):
  """
  获取CPU信息。
  """
  print
  print "cpu_info:"
  if os == "Windows":
    tmpdict = {}
    tmpdict["CpuCores"] = 0
    c = wmi.WMI()
    for cpu in c.Win32_Processor():
      tmpdict["CpuType"] = cpu.Name
    try:
      tmpdict["CpuCores"] = cpu.NumberOfCores
    except:
      tmpdict["CpuCores"] += 1
      tmpdict["CpuClock"] = cpu.MaxClockSpeed
    print '\t' + 'CpuType :\t' + str(tmpdict["CpuType"])
    print '\t' + 'CpuCores :\t' + str(tmpdict["CpuCores"])


def get_network_info(os):
  """
  获取网卡信息和当前TCP连接数。
  """
  print
  print "network_info:"
  if os == "Windows":
    tmplist = []
    c = wmi.WMI()
    for interface in c.Win32_NetworkAdapterConfiguration(IPEnabled=1):
      tmpdict = {}
      tmpdict["Description"] = interface.Description
      tmpdict["IPAddress"] = interface.IPAddress[0]
      tmpdict["IPSubnet"] = interface.IPSubnet[0]
      tmpdict["MAC"] = interface.MACAddress
      tmplist.append(tmpdict)
    for i in tmplist:
      print '\t' + i["Description"]
      print '\t' + '\t' + "MAC :" + '\t' + i["MAC"]
      print '\t' + '\t' + "IPAddress :" + '\t' + i["IPAddress"]
      print '\t' + '\t' + "IPSubnet :" + '\t' + i["IPSubnet"]
    for interfacePerfTCP in c.Win32_PerfRawData_Tcpip_TCPv4():
      print '\t' + 'TCP Connect :\t' + str(interfacePerfTCP.ConnectionsEstablished)


if __name__ == "__main__":
  os = platform.system()
  get_system_info(os)
  get_memory_info(os)
  get_disk_info(os)
  get_cpu_info(os)
  get_network_info(os)